package system

import (
  "encoding/json"
  "errors"
  "fmt"
  "io/ioutil"
  "livefind/global"
  "livefind/model/system"
  "livefind/model/system/request"
  "livefind/utils"
  "os"
  "path/filepath"
  "strconv"
  "strings"
  "text/template"

  "gorm.io/gorm"
)

const (
  autoPath = "autocode_template/"
  basePath = "resource/template"
)

var injectionPaths []injectionMeta

func Init() {
  if len(injectionPaths) != 0 {
    return
  }
  injectionPaths = []injectionMeta{
    {
      path: filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SInitialize, "gorm.go"),
      funcName:    "MysqlTables",
      structNameF: "autocode.%s{},",
    },
    {
      path: filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SInitialize, "router.go"),
      funcName:    "Routers",
      structNameF: "autocodeRouter.Init%sRouter(PrivateGroup)",
    },
    {
      path: filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SApi, "enter.go"),
      funcName:    "ApiGroup",
      structNameF: "%sApi",
    },
    {
      path: filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SRouter, "enter.go"),
      funcName:    "RouterGroup",
      structNameF: "%sRouter",
    },
    {
      path: filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SService, "enter.go"),
      funcName:    "ServiceGroup",
      structNameF: "%sService",
    },
  }
}

type injectionMeta struct {
  path        string
  funcName    string
  structNameF string // 带格式化的
}

type tplData struct {
  template         *template.Template
  locationPath     string
  autoCodePath     string
  autoMoveFilePath string
}

type AutoCodeService struct {
}

var AutoCodeServiceApp = new(AutoCodeService)

//@author: [songzhibin97](https://github.com/songzhibin97)
//@function: PreviewTemp
//@description: 预览创建代码
//@param: model.AutoCodeStruct
//@return: map[string]string, error

func (autoCodeService *AutoCodeService) PreviewTemp(autoCode system.AutoCodeStruct) (map[string]string, error) {
  dataList, _, needMkdir, err := autoCodeService.getNeedList(&autoCode)
  if err != nil {
    return nil, err
  }

  // 写入文件前，先创建文件夹
  if err = utils.CreateDir(needMkdir...); err != nil {
    return nil, err
  }

  // 创建map
  ret := make(map[string]string)

  // 生成map
  for _, value := range dataList {
    ext := ""
    if ext = filepath.Ext(value.autoCodePath); ext == ".txt" {
      continue
    }
    f, err := os.OpenFile(value.autoCodePath, os.O_CREATE|os.O_WRONLY, 0755)
    if err != nil {
      return nil, err
    }
    if err = value.template.Execute(f, autoCode); err != nil {
      return nil, err
    }
    _ = f.Close()
    f, err = os.OpenFile(value.autoCodePath, os.O_CREATE|os.O_RDONLY, 0755)
    if err != nil {
      return nil, err
    }
    builder := strings.Builder{}
    builder.WriteString("```")

    if ext != "" && strings.Contains(ext, ".") {
      builder.WriteString(strings.Replace(ext, ".", "", -1))
    }
    builder.WriteString("\n\n")
    data, err := ioutil.ReadAll(f)
    if err != nil {
      return nil, err
    }
    builder.Write(data)
    builder.WriteString("\n\n```")

    pathArr := strings.Split(value.autoCodePath, string(os.PathSeparator))
    ret[pathArr[1]+"-"+pathArr[3]] = builder.String()
    _ = f.Close()

  }
  defer func() { // 移除中间文件
    if err := os.RemoveAll(autoPath); err != nil {
      return
    }
  }()
  return ret, nil
}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: CreateTemp
//@description: 创建代码
//@param: model.AutoCodeStruct
//@return: err error

func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruct, ids ...uint) (err error) {
  dataList, fileList, needMkdir, err := autoCodeService.getNeedList(&autoCode)
  if err != nil {
    return err
  }
  meta, _ := json.Marshal(autoCode)
  // 写入文件前，先创建文件夹
  if err = utils.CreateDir(needMkdir...); err != nil {
    return err
  }

  // 生成文件
  for _, value := range dataList {
    f, err := os.OpenFile(value.autoCodePath, os.O_CREATE|os.O_WRONLY, 0755)
    if err != nil {
      return err
    }
    if err = value.template.Execute(f, autoCode); err != nil {
      return err
    }
    _ = f.Close()
  }

  defer func() { // 移除中间文件
    if err := os.RemoveAll(autoPath); err != nil {
      return
    }
  }()
  bf := strings.Builder{}
  idBf := strings.Builder{}
  injectionCodeMeta := strings.Builder{}
  for _, id := range ids {
    idBf.WriteString(strconv.Itoa(int(id)))
    idBf.WriteString(";")
  }
  if autoCode.AutoMoveFile { // 判断是否需要自动转移
    Init()
    for index := range dataList {
      autoCodeService.addAutoMoveFile(&dataList[index])
    }
    for _, value := range dataList { // 移动文件
      if err := utils.FileMove(value.autoCodePath, value.autoMoveFilePath); err != nil {
        return err
      }
    }
    err = injectionCode(autoCode.StructName, &injectionCodeMeta)
    if err != nil {
      return
    }
    // 保存生成信息
    for _, data := range dataList {
      if len(data.autoMoveFilePath) != 0 {
        bf.WriteString(data.autoMoveFilePath)
        bf.WriteString(";")
      }
    }

    if global.GVA_CONFIG.AutoCode.TransferRestart {
      go func() {
        _ = utils.Reload()
      }()
    }
  } else { // 打包
    if err = utils.ZipFiles("./ginvueadmin.zip", fileList, ".", "."); err != nil {
      return err
    }
  }
  if autoCode.AutoMoveFile || autoCode.AutoCreateApiToSql {
    if autoCode.TableName != "" {
      err = AutoCodeHistoryServiceApp.CreateAutoCodeHistory(
        string(meta),
        autoCode.StructName,
        autoCode.Description,
        bf.String(),
        injectionCodeMeta.String(),
        autoCode.TableName,
        idBf.String(),
      )
    } else {
      err = AutoCodeHistoryServiceApp.CreateAutoCodeHistory(
        string(meta),
        autoCode.StructName,
        autoCode.Description,
        bf.String(),
        injectionCodeMeta.String(),
        autoCode.StructName,
        idBf.String(),
      )
    }
  }
  if err != nil {
    return err
  }
  if autoCode.AutoMoveFile {
    return errors.New("创建代码成功并移动文件成功")
  }
  return nil

}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetAllTplFile
//@description: 获取 pathName 文件夹下所有 tpl 文件
//@param: pathName string, fileList []string
//@return: []string, error

func (autoCodeService *AutoCodeService) GetAllTplFile(pathName string, fileList []string) ([]string, error) {
  files, err := ioutil.ReadDir(pathName)
  for _, fi := range files {
    if fi.IsDir() {
      fileList, err = autoCodeService.GetAllTplFile(pathName+"/"+fi.Name(), fileList)
      if err != nil {
        return nil, err
      }
    } else {
      if strings.HasSuffix(fi.Name(), ".tpl") {
        fileList = append(fileList, pathName+"/"+fi.Name())
      }
    }
  }
  return fileList, err
}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetTables
//@description: 获取数据库的所有表名
//@param: dbName string
//@return: err error, TableNames []request.TableReq

func (autoCodeService *AutoCodeService) GetTables(dbName string) (err error, TableNames []request.TableReq) {
  err = global.GVA_DB.Raw("select table_name as table_name from information_schema.tables where table_schema = ?", dbName).Scan(&TableNames).Error
  return err, TableNames
}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetDB
//@description: 获取数据库的所有数据库名
//@return: err error, DBNames []request.DBReq

func (autoCodeService *AutoCodeService) GetDB() (err error, DBNames []request.DBReq) {
  err = global.GVA_DB.Raw("SELECT SCHEMA_NAME AS `database` FROM INFORMATION_SCHEMA.SCHEMATA;").Scan(&DBNames).Error
  return err, DBNames
}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: GetDB
//@description: 获取指定数据库和指定数据表的所有字段名,类型值等
//@param: tableName string, dbName string
//@return: err error, Columns []request.ColumnReq

func (autoCodeService *AutoCodeService) GetColumn(tableName string, dbName string) (err error, Columns []request.ColumnReq) {
  err = global.GVA_DB.Raw("SELECT COLUMN_NAME column_name,DATA_TYPE data_type,CASE DATA_TYPE WHEN 'longtext' THEN c.CHARACTER_MAXIMUM_LENGTH WHEN 'varchar' THEN c.CHARACTER_MAXIMUM_LENGTH WHEN 'double' THEN CONCAT_WS( ',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE ) WHEN 'decimal' THEN CONCAT_WS( ',', c.NUMERIC_PRECISION, c.NUMERIC_SCALE ) WHEN 'int' THEN c.NUMERIC_PRECISION WHEN 'bigint' THEN c.NUMERIC_PRECISION ELSE '' END AS data_type_long,COLUMN_COMMENT column_comment FROM INFORMATION_SCHEMA.COLUMNS c WHERE table_name = ? AND table_schema = ?", tableName, dbName).Scan(&Columns).Error
  return err, Columns
}

func (autoCodeService *AutoCodeService) DropTable(tableName string) error {
  return global.GVA_DB.Exec("DROP TABLE " + tableName).Error
}

//@author: [SliverHorn](https://github.com/SliverHorn)
//@author: [songzhibin97](https://github.com/songzhibin97)
//@function: addAutoMoveFile
//@description: 生成对应的迁移文件路径
//@param: *tplData
//@return: null

func (autoCodeService *AutoCodeService) addAutoMoveFile(data *tplData) {
  base := filepath.Base(data.autoCodePath)
  fileSlice := strings.Split(data.autoCodePath, string(os.PathSeparator))
  n := len(fileSlice)
  if n <= 2 {
    return
  }
  if strings.Contains(fileSlice[1], "server") {
    if strings.Contains(fileSlice[n-2], "router") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root, global.GVA_CONFIG.AutoCode.Server,
        global.GVA_CONFIG.AutoCode.SRouter, base)
    } else if strings.Contains(fileSlice[n-2], "api") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SApi, base)
    } else if strings.Contains(fileSlice[n-2], "service") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SService, base)
    } else if strings.Contains(fileSlice[n-2], "model") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SModel, base)
    } else if strings.Contains(fileSlice[n-2], "request") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Server, global.GVA_CONFIG.AutoCode.SRequest, base)
    }
  } else if strings.Contains(fileSlice[1], "web") {
    if strings.Contains(fileSlice[n-1], "js") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Web, global.GVA_CONFIG.AutoCode.WApi, base)
    } else if strings.Contains(fileSlice[n-2], "form") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Web, global.GVA_CONFIG.AutoCode.WForm, filepath.Base(filepath.Dir(filepath.Dir(data.autoCodePath))), strings.TrimSuffix(base, filepath.Ext(base))+"Form.vue")
    } else if strings.Contains(fileSlice[n-2], "table") {
      data.autoMoveFilePath = filepath.Join(global.GVA_CONFIG.AutoCode.Root,
        global.GVA_CONFIG.AutoCode.Web, global.GVA_CONFIG.AutoCode.WTable, filepath.Base(filepath.Dir(filepath.Dir(data.autoCodePath))), base)
    }
  }
}

//@author: [piexlmax](https://github.com/piexlmax)
//@author: [SliverHorn](https://github.com/SliverHorn)
//@function: CreateApi
//@description: 自动创建api数据,
//@param: a *model.AutoCodeStruct
//@return: err error

func (autoCodeService *AutoCodeService) AutoCreateApi(a *system.AutoCodeStruct) (ids []uint, err error) {
  var apiList = []system.SysApi{
    {
      Path:        "/" + a.Abbreviation + "/" + "create" + a.StructName,
      Description: "新增" + a.Description,
      ApiGroup:    a.Abbreviation,
      Method:      "POST",
    },
    {
      Path:        "/" + a.Abbreviation + "/" + "delete" + a.StructName,
      Description: "删除" + a.Description,
      ApiGroup:    a.Abbreviation,
      Method:      "DELETE",
    },
    {
      Path:        "/" + a.Abbreviation + "/" + "delete" + a.StructName + "ByIds",
      Description: "批量删除" + a.Description,
      ApiGroup:    a.Abbreviation,
      Method:      "DELETE",
    },
    {
      Path:        "/" + a.Abbreviation + "/" + "update" + a.StructName,
      Description: "更新" + a.Description,
      ApiGroup:    a.Abbreviation,
      Method:      "PUT",
    },
    {
      Path:        "/" + a.Abbreviation + "/" + "find" + a.StructName,
      Description: "根据ID获取" + a.Description,
      ApiGroup:    a.Abbreviation,
      Method:      "GET",
    },
    {
      Path:        "/" + a.Abbreviation + "/" + "get" + a.StructName + "List",
      Description: "获取" + a.Description + "列表",
      ApiGroup:    a.Abbreviation,
      Method:      "GET",
    },
  }
  err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {

    for _, v := range apiList {
      var api system.SysApi
      if errors.Is(tx.Where("path = ? AND method = ?", v.Path, v.Method).First(&api).Error, gorm.ErrRecordNotFound) {
        if err = tx.Create(&v).Error; err != nil { // 遇到错误时回滚事务
          return err
        } else {
          ids = append(ids, v.ID)
        }
      }
    }
    return nil
  })
  return ids, err
}

func (autoCodeService *AutoCodeService) getNeedList(autoCode *system.AutoCodeStruct) (dataList []tplData, fileList []string, needMkdir []string, err error) {
  // 去除所有空格
  utils.TrimSpace(autoCode)
  for _, field := range autoCode.Fields {
    utils.TrimSpace(field)
  }
  // 获取 basePath 文件夹下所有tpl文件
  tplFileList, err := autoCodeService.GetAllTplFile(basePath, nil)
  if err != nil {
    return nil, nil, nil, err
  }
  dataList = make([]tplData, 0, len(tplFileList))
  fileList = make([]string, 0, len(tplFileList))
  needMkdir = make([]string, 0, len(tplFileList)) // 当文件夹下存在多个tpl文件时，改为map更合理
  // 根据文件路径生成 tplData 结构体，待填充数据
  for _, value := range tplFileList {
    dataList = append(dataList, tplData{locationPath: value})
  }
  // 生成 *Template, 填充 template 字段
  for index, value := range dataList {
    dataList[index].template, err = template.ParseFiles(value.locationPath)
    if err != nil {
      return nil, nil, nil, err
    }
  }
  // 生成文件路径，填充 autoCodePath 字段，readme.txt.tpl不符合规则，需要特殊处理
  // resource/template/web/api.js.tpl -> autoCode/web/autoCode.PackageName/api/autoCode.PackageName.js
  // resource/template/readme.txt.tpl -> autoCode/readme.txt
  for index, value := range dataList {
    trimBase := strings.TrimPrefix(value.locationPath, basePath+"/")
    if trimBase == "readme.txt.tpl" {
      dataList[index].autoCodePath = autoPath + "readme.txt"
      continue
    }

    if lastSeparator := strings.LastIndex(trimBase, "/"); lastSeparator != -1 {
      origFileName := strings.TrimSuffix(trimBase[lastSeparator+1:], ".tpl")
      firstDot := strings.Index(origFileName, ".")
      if firstDot != -1 {
        var fileName string
        if origFileName[firstDot:] != ".go" {
          fileName = autoCode.PackageName + origFileName[firstDot:]
        } else {
          fileName = autoCode.HumpPackageName + origFileName[firstDot:]
        }

        dataList[index].autoCodePath = filepath.Join(autoPath, trimBase[:lastSeparator], autoCode.PackageName,
          origFileName[:firstDot], fileName)
      }
    }

    if lastSeparator := strings.LastIndex(dataList[index].autoCodePath, string(os.PathSeparator)); lastSeparator != -1 {
      needMkdir = append(needMkdir, dataList[index].autoCodePath[:lastSeparator])
    }
  }
  for _, value := range dataList {
    fileList = append(fileList, value.autoCodePath)
  }
  return dataList, fileList, needMkdir, err
}

// injectionCode 封装代码注入
func injectionCode(structName string, bf *strings.Builder) error {
  for _, meta := range injectionPaths {
    code := fmt.Sprintf(meta.structNameF, structName)
    if err := utils.AutoInjectionCode(meta.path, meta.funcName, code); err != nil {
      return err
    }
    bf.WriteString(fmt.Sprintf("%s@%s@%s;", meta.path, meta.funcName, code))
  }
  return nil
}
